package edu.stanford.bmir.protege.web.client.ui.ontology.stkos.enricher.common;

import com.gwtext.client.widgets.tree.TreeNode;

import edu.stanford.bmir.protege.web.client.project.Project;
import edu.stanford.bmir.protege.web.client.rpc.data.EntityData;
import edu.stanford.bmir.protege.web.client.ui.ontology.stkos.common.treepanel.StkosPropertyTreePanel;

public class StkosEnricherPropertyTreePanel extends StkosPropertyTreePanel {

	public StkosEnricherPropertyTreePanel(Project project) {
		super(project);
	}

	@Override
	protected TreeNode createTreeNode(EntityData entityData) {
		TreeNode node = super.createTreeNode(entityData);
		if (node == null) {
			return node;
		}

		addAnnotationEnrichedFlagNodeMark(node, entityData);
		return node;
	}
}
